package simple;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

/**
 * 假设 Andy 和 Doris 想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。
 *
 * 你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。
 * 你可以假设答案总是存在。
 *
 * 链接：https://leetcode.cn/problems/minimum-index-sum-of-two-lists
 *
 * @author 胡宇轩
 * @Email: programboy@163.com
 */
public class MinimumIndexSumOfTwoLists {
    class Solution {
        public String[] findRestaurant(String[] list1, String[] list2) {
            int minIndex = Integer.MAX_VALUE;
            List<String> result = new ArrayList<>();
            HashMap<String, Integer> map = new HashMap<>(list1.length);

            for (int i = 0; i < list1.length; i++) {
                map.put(list1[i], i);
            }

            for (int i = 0; i < list2.length; i++) {
                if(map.containsKey(list2[i])){
                    int index = map.get(list2[i]) + i;
                    if(index < minIndex){
                        result.clear();
                        minIndex = index;
                    }
                    if(index <= minIndex){
                        result.add(list2[i]);
                    }
                }
            }
            return result.toArray(new String[0]);
        }
    }
}
